package com.teamsun.datasource;

/**
 * <p>动态设置访问数据源名称</p>
 */
public class DatasourceHolder {
    private static final ThreadLocal<String> dataSource = new ThreadLocal<String>();

    /**
     * 设置数据源名称
     * @param dbName 数据源名称
     */
    public static void setDBName(String dbName) {
        dataSource.set(dbName);
    }

    /**
     * 获取数据源名称
     * @return
     */
    public static String getDBName() {
        return dataSource.get();
    }

    public static void remove() {
        dataSource.remove();
    }
}
